reshape()参数固定太死，导致dataloader最后一次迭代获得结果不满足reshape()要求，应该reshape(-1,...)_shapeisinvalidforinput...

浏览 50 次 标签： PyTorch报错shape ‘[16    28    28]‘ is invalid for input of size 6272?尝试设置-1